import json

def get_pseudo_code_from_model_res(input:str):
    if input.find("</think>") != -1:
        input = input[(input.find("</think>") + 8):]
    if input.find("```json") != -1:
        input = input[(input.find("```json") + 7):]
    if input.rfind("```") != -1:
        input = input[:input.rfind("```")]
    try:
        return json.loads(input.strip())["CODE"]
    except Exception as e:
        return ""


input = r"""
```json
{
    "REASONING": "### Step[1]: 检查[伪代码书写格式规范]\n1. 检查控制流语句的书写格式\n- 原始伪代码中的控制流语句没有按照规范使用`IF1`和`ENDIF1`进行包裹，且条件判断符号`=`左右缺少空格。\n2. 检查注释和自然语言描述\n- 原始伪代码中没有注释，但自然语言描述需要按照规范进行改写。\n3. 检查语句结尾的分号\n- 原始伪代码中部分语句缺少分号结尾，需要补充。\n4. 检查条件判断符号的空格\n- 原始伪代码中条件判断符号`=`左右缺少空格，需要补充。\n5. 检查公式的书写格式\n- 原始伪代码中的公式使用了`$$`包裹，符合规范，无需修改。",
    "CODE": "IF1(Type == 0x00)\n$$y = uu;$$\nELSEIF1(Type == 0x0F)\n$$x1 = A1 \\bullet x1 + H1 \\bullet u + H1 \\bullet y \\bullet y;$$\n$$y = x1 + D \\bullet u;$$\nELSEIF1(Type == 0xF0)\n$$Filter_{Temp} = x1 \\bullet x3;$$\n$$x1 = x2 + H1 \\bullet u;$$\n$$x2 = A1 \\bullet Filter_{Temp} + A2 \\bullet x2 + H2 \\bullet u;$$\n$$y = x1 + D \\bullet u;$$\nENDIF1"
}
```
"""
if __name__ == "__main__":
    get_pseudo_code_from_model_res(input)